/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/

#include "itkSparseFrequencyContainer2.h"


int itkSparseFrequencyContainer2Test(int, char* [] )
{
  std::cout << "SparseFrequencyContainer2 Test \n \n";

  using SparseFrequencyContainer2Type = itk::Statistics::SparseFrequencyContainer2;


  SparseFrequencyContainer2Type::Pointer container =
                                            SparseFrequencyContainer2Type::New();

  using AbsoluteFrequencyType = SparseFrequencyContainer2Type::AbsoluteFrequencyType;

  constexpr unsigned int numberOfBins = 1250;

  container->Initialize( numberOfBins );


  // Test the SetFrequency() / GetFrequency() methods
    {
    std::cout << "Testing Set/Get Frequency methods...";
    for( unsigned int bin=0; bin < numberOfBins; bin++ )
      {
      // Compute any value as frequency just to test the SetFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>( bin * bin );
      container->SetFrequency( bin, frequency );
      }

    for( unsigned int bin=0; bin < numberOfBins; bin++ )
      {
      // Test if the values can be read back
      const auto frequency = static_cast<AbsoluteFrequencyType>( bin * bin );
      const AbsoluteFrequencyType stored    = container->GetFrequency( bin );
      if( stored != frequency )
        {
        std::cout << "Failed !" << std::endl;
        std::cout << "Stored Frequency in bin " << bin << " doesn't match value" << std::endl;
        std::cout << "Value is = " << stored << " value should be " << frequency << std::endl;
        return EXIT_FAILURE;
        }
      }
    std::cout << " PASSED !" << std::endl;

    }   // end of SetFrequency() / GetFrequency() test

  //Set all the bins to zero and check the values
  container->SetToZero();
  for( unsigned int bin=0; bin < numberOfBins; bin++ )
    {
    if( container->GetFrequency( bin ) != itk::NumericTraits< AbsoluteFrequencyType >::ZeroValue() )
      {
      std::cout << "Failed !" << std::endl;
      std::cout << "Stored Frequency in bin is not zero after SetToZero() method invocation"
                << std::endl;
      return EXIT_FAILURE;
      }
    }

  // Test the IncreaseFrequency() method
    {
    std::cout << "Testing IncreaseFrequency method...";
    // Try not to depend on previous tests....So, we initialize the histogram again.
    for( unsigned int bin=0; bin < numberOfBins; bin++ )
      {
      // Compute any value as frequency just to test the SetFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>( bin * bin );
      container->SetFrequency( bin, frequency );
      }

    // Now increment by a number (we use "bin", but any other will do it...)
    for( unsigned int bin=0; bin < numberOfBins; bin++ )
      {
      // Compute any value as frequency just to test the IncreaseFrequency() method
      const auto frequency = static_cast<AbsoluteFrequencyType>( bin );
      container->IncreaseFrequency( bin, frequency );
      }


    // Test if the values can be read back
    for( unsigned int bin=0; bin < numberOfBins; bin++ )
      {
      const auto frequency = static_cast<AbsoluteFrequencyType>( bin * bin + bin );
      const AbsoluteFrequencyType stored    = container->GetFrequency( bin );
      if( stored != frequency )
        {
        std::cout << "Failed !" << std::endl;
        std::cout << "Stored Frequency in bin " << bin << " doesn't match value" << std::endl;
        std::cout << "Value is = " << stored << " value should be " << frequency << std::endl;
        return EXIT_FAILURE;
        }
      }
    std::cout << " PASSED !" << std::endl;
    }   // end of SetFrequency() / GetFrequency() test

  std::cout << "Test passed." << std::endl;
  return EXIT_SUCCESS;

}
